#!/bin/bash
#SBATCH --gpus=1

module load cuda/11.7
module load cudnn/8.2.1_CUDA11.x
module load gcc/7.3
source activate regionclip

# export http_proxy=http://192.168.1.23:8888
# export https_proxy=http://192.168.1.23:8888
# export ftp_proxy=ftp://192.168.1.23:8888


# export CUDA_HOME=/data/apps/cuda/10.1
# export PATH=/data/home/scv6140/run/1/hip/bin:$PATH

# cd ./models/ops
# sh ./make.sh
# # Task 1
python3 ./tools/train_net.py \
--num-gpus 4  \
--config-file ./configs/COCO-InstanceSegmentation/CLIP_fast_rcnn_R_50_C4_cityscapes.yaml  \
MODEL.WEIGHTS ./pretrained_ckpt/regionclip/regionclip_pretrained-cc_rn50.pth  \
MODEL.CLIP.OFFLINE_RPN_CONFIG ./configs/COCO-InstanceSegmentation/mask_rcnn_R_50_C4_1x_ovd_FSD_cityscapes.yaml \
MODEL.CLIP.BB_RPN_WEIGHTS ./pretrained_ckpt/rpn/rpn_coco_80.pth  \
MODEL.CLIP.TEXT_EMB_PATH ./pretrained_ckpt/concept_emb/concept_embeds_8.pth \
MODEL.CLIP.OPENSET_TEST_TEXT_EMB_PATH ./pretrained_ckpt/concept_emb/concept_embeds_8.pth  \
OUTPUT_DIR ./logs/cityscapes

